package com.jiyinit.web.core.common.util;

import net.sourceforge.pinyin4j.PinyinHelper;
import net.sourceforge.pinyin4j.format.HanyuPinyinOutputFormat;
import net.sourceforge.pinyin4j.format.HanyuPinyinToneType;
import net.sourceforge.pinyin4j.format.exception.BadHanyuPinyinOutputFormatCombination;

/**
 * Created by lr on 2018/1/22.
 */
public class PinyinUtil {
    public static  HanyuPinyinOutputFormat format = null;
    public static String[] pinyin = null;

    static{
        format = new HanyuPinyinOutputFormat();
        format.setToneType(HanyuPinyinToneType.WITHOUT_TONE);
    }

    /**
     * 转换单个字符
     * @param c
     * @return
     */
    public static String trans2Char(char c){
        try{
            pinyin = PinyinHelper.toHanyuPinyinStringArray(c, format);
        }catch (BadHanyuPinyinOutputFormatCombination e){
            e.printStackTrace();
        }
        // 如果c不是汉字，toHanyuPinyinStringArray会返回null
        // 只取一个发音，如果是多音字，仅取第一个发音
        return (pinyin == null) ? null : pinyin[0];
    }

    /**
     * 转换一段字符串为全拼
     * @param str
     * @return
     */
    public  static String trans2QuanPin(String str){
        StringBuffer sb = new StringBuffer();
        String tempPinyin = null;
        for (int i = 0; i < str.length(); i++){
            tempPinyin = trans2Char(str.charAt(i));
            if (tempPinyin == null){ // 如果str.charAt(i)非汉字，则保持原样
                sb.append(str.charAt(i));
            }else{
                sb.append(tempPinyin);
            }
        }
        return sb.toString();
    }

    /**
     * 转换成 汉字首字母
     * @param str
     * @return
     */
    public static String trans2HeadChar(String str){
        StringBuffer sb = new StringBuffer();
        for (int i = 0; i < str.length(); i++) {
            String tempPinyin[] = PinyinHelper.toHanyuPinyinStringArray(str.charAt(i));
            if (tempPinyin == null){ // 如果str.charAt(i)非汉字，则保持原样
                sb.append(str.charAt(i));
            }else{// 提取汉字的首字母
                sb.append(tempPinyin[0].charAt(0));
            }
        }
        return sb.toString();
    }

    public static void main(String[] args) {
        System.out.println(PinyinUtil.trans2HeadChar("哈哈哈，这是一个好的good idea 吧？1000块钱值了吧？"));
        System.out.println(PinyinUtil.trans2QuanPin("哈哈哈，这是一个好的good idea 吧？1000块钱值了吧？"));
    }
}
